# 定义GPU数量变量
NUM_GPUS=2

#训练代码
python train_net_lmpm.py \
    --config-file configs/tlggnv4_SWIN_bs8.yaml \
    --num-gpus $NUM_GPUS --dist-url auto \
    --tlggn_k 3 \
    --resume \
    MODEL.WEIGHTS ./weights/model_final_86143f.pkl \
    SOLVER.IMS_PER_BATCH $NUM_GPUS \
    OUTPUT_DIR ./outputs/tlggnv4k3

# 检查训练是否成功完成
if [ $? -eq 0 ]; then
    echo "训练完成，开始推理..."
    #推理代码
CUDA_VISIBLE_DEVICES=2,3 python train_net_lmpm.py \
        --config-file configs/tlggnv4_SWIN_bs8.yaml \
        --num-gpus $NUM_GPUS --dist-url auto --eval-only \
        --tlggn_k 3 \
        MODEL.WEIGHTS ./outputs/tlggnv4k3/model_final.pth \
        SOLVER.IMS_PER_BATCH $NUM_GPUS \
        OUTPUT_DIR ./outputs/tlggnv4k3
else
    echo "训练失败，推理未执行。"
fi 